/**
给定一个整数数组和一个整数 k ，请找到该数组中和为 k 的连续子数组的个数。

示例 1：
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况

示例 2：
输入:nums = [1,2,3], k = 3
输出: 2

提示:

1 <= nums.length <= 2 * 10e4
-1000 <= nums[i] <= 1000
-10e7 <= k <= 10e7

*/

#pragma once

#include <algorithm>
#include <vector>

using namespace std;

class Solution {
   public:
    int subarraySum(vector<int>& nums, int k) {
        auto s = k;
        int ns = 0;
        for (auto n : nums) {
            s -= n;
            if (s == 0) {
                ns += 1;
                s = k;
            } else if (s < 0) {
                s = k;
            }
        }
        return ns;
    }
};

int answer(vector<int> nums, int k) {
    auto a = Solution();
    return a.subarraySum(nums, k);
}
